Skip to content

Fix PubChem pretraining#159

Closed
sfluegel05 wants to merge 5 commits intodevfrom
fix/pubchem-pretraining
Closed

Fix PubChem pretraining#159
sfluegel05 wants to merge 5 commits intodevfrom
fix/pubchem-pretraining

Conversation

@sfluegel05
Copy link
Collaborator

Due to various changes in other parts of chebai, Pretraining with PubChem fails. This PR includes some fixes:

  • models are allowed to have input and output dimensions of 0
  • a label file is only expected for DynamicDataset subclasses
  • the missing_labels property gets set to false if there are no labels

@sfluegel05 sfluegel05 requested a review from schnamo March 4, 2026 13:22
@schnamo
Copy link
Collaborator

schnamo commented Mar 5, 2026

That seems to work, thank you!
Following up on this, I will adjust the default vocab size in the config files.

@schnamo schnamo closed this Mar 5, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants